greedy *800

Please click on ads to support us..

Python Code:

t = int(input())
for k in range(t):
    n = int(input())
    a = list(map(int, input().split(' ')))
    s = sum(a)
    if(s % n == 0 and s//n in a):
        print('YES')
    else:
        print('NO')
            
            

C++ Code:

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
#define fi first
#define se second
#define rep(i,a,n) for(int i=a;i<n;i++)
#define per(i,a,n) for(int i=n-1;i>=a;i--)
#define pb push_back
#define yes {cout<<"Yes"<<endl;return;}
#define no {cout<<"No"<<endl;return;}
using namespace std;
const int mod = 1e9+7;
int a[55];
void solve(){
    int n;cin>>n;
    int sum=0;rep(i,1,n+1){
        cin>>a[i];sum+=a[i];
    }
    rep(i,1,n+1){
        if(sum-a[i]==(n-1)*a[i]){
            yes
        }
    }
    no
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t=1;cin>>t;
    while(t--){
        solve();
    }
}


Comments

Submit
0 Comments
More Questions

1487A - Arena
1520D - Same Differences
376A - Lever
1305A - Kuroni and the Gifts
1609A - Divide and Multiply
149B - Martian Clock
205A - Little Elephant and Rozdil
1609B - William the Vigilant
978B - File Name
1426B - Symmetric Matrix
732B - Cormen --- The Best Friend Of a Man
1369A - FashionabLee
1474B - Different Divisors
1632B - Roof Construction
388A - Fox and Box Accumulation
451A - Game With Sticks
768A - Oath of the Night's Watch
156C - Cipher
545D - Queue
459B - Pashmak and Flowers
1538A - Stone Game
1454C - Sequence Transformation
165B - Burning Midnight Oil
17A - Noldbach problem
1350A - Orac and Factors
1373A - Donut Shops
26A - Almost Prime
1656E - Equal Tree Sums
1656B - Subtract Operation
1656A - Good Pairs